from qtpy import QtWidgets


class ColorDialog(QtWidgets.QColorDialog):
    """颜色对话框"""
    def __init__(self, parent=None):
        super(ColorDialog, self).__init__(parent)
        self.setOption(QtWidgets.QColorDialog.ShowAlphaChannel)
        # Mac本机对话框不支持我们的还原按钮。
        self.setOption(QtWidgets.QColorDialog.DontUseNativeDialog)
        # 添加恢复默认值按钮。
        # 默认值是在调用时设置的，因此它可以在不同元素的对话框之间工作。
        self.default = None
        self.bb = self.layout().itemAt(1).widget()
        self.bb.addButton(QtWidgets.QDialogButtonBox.RestoreDefaults)
        self.bb.clicked.connect(self.checkRestore)

    def getColor(self, value=None, title=None, default=None):
        """
        获取颜色对话框
        :param value 需要设置给对话框的颜色值
        :param title 对话框标题
        :param default 默认值
        """
        self.default = default
        if title:
            self.setWindowTitle(title)
        if value:
            self.setCurrentColor(value)
        return self.currentColor() if self.exec_() else None

    def checkRestore(self, button):
        """
        选中恢复
        :param button 选择的回复按钮
        """
        if (
            self.bb.buttonRole(button) & QtWidgets.QDialogButtonBox.ResetRole
            and self.default
        ):
            self.setCurrentColor(self.default)
